Domain-Specific Multi-Level IR Rewriting for GPU

Most compilers have a single core intermediate representation (IR) (e.g., LLVM) sometimes complemented with vaguely defined IR-like data structures. This IR is commonly low-level and close to machine instructions. As result, optimizations relying on domain-specific information are either not possible or require complex analysis recover the missing information. In contrast, multi-level rewriting instantiates hierarchy of dialects (IRs), lowers programs level-by-level, performs code transformations at most suitable level. We demonstrate effectiveness this approach for weather climate domain. particular, we develop prototype compiler design stencil- GPU-specific based set newly introduced principles. find that two (500 lines code) realized top LLVM’s extensible MLIR infrastructure suffice outperform state-of-the-art solutions. essence, promises herald age specialized composed from domain- target-specific implemented shared infrastructure.